/*
 * Copyright [2022] [https://www.xiaonuo.vip]
 *
 * Snowy采用APACHE LICENSE 2.0开源协议，您在使用过程中，需要注意以下几点：
 *
 * 1.请不要删除和修改根目录下的LICENSE文件。
 * 2.请不要删除和修改Snowy源码头部的版权声明。
 * 3.本项目代码可免费商业使用，商业使用请保留源码和相关描述文件的项目出处，作者声明等。
 * 4.分发源码时候，请注明软件出处 https://www.xiaonuo.vip
 * 5.不可二次分发开源参与同类竞品，如有想法可联系团队xiaonuobase@qq.com商议合作。
 * 6.若您的项目无法满足以上几点，需要更多功能代码，获取Snowy商业授权许可，请在官网购买授权，地址为 https://www.xiaonuo.vip
 */
package org.wenshu.ai.modular.datasource.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollStreamUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.wenshu.common.enums.CommonSortOrderEnum;
import org.wenshu.common.exception.CommonException;
import org.wenshu.common.page.CommonPageRequest;
import org.wenshu.ai.modular.datasource.entity.AiDatasourceConfig;
import org.wenshu.ai.modular.datasource.mapper.AiDatasourceConfigMapper;
import org.wenshu.ai.modular.datasource.param.AiDatasourceConfigAddParam;
import org.wenshu.ai.modular.datasource.param.AiDatasourceConfigEditParam;
import org.wenshu.ai.modular.datasource.param.AiDatasourceConfigIdParam;
import org.wenshu.ai.modular.datasource.param.AiDatasourceConfigPageParam;
import org.wenshu.ai.modular.datasource.service.AiDatasourceConfigService;

import java.util.List;

/**
 * 数据源配置表Service接口实现类
 *
 * @author pok
 * @date  2025/03/12 14:21
 **/
@Service
public class AiDatasourceConfigServiceImpl extends ServiceImpl<AiDatasourceConfigMapper, AiDatasourceConfig> implements AiDatasourceConfigService {

    @Override
    public Page<AiDatasourceConfig> page(AiDatasourceConfigPageParam aiDatasourceConfigPageParam) {
        QueryWrapper<AiDatasourceConfig> queryWrapper = new QueryWrapper<AiDatasourceConfig>().checkSqlInjection();
        if(ObjectUtil.isNotEmpty(aiDatasourceConfigPageParam.getName())) {
            queryWrapper.lambda().like(AiDatasourceConfig::getName, aiDatasourceConfigPageParam.getName());
        }
        if(ObjectUtil.isNotEmpty(aiDatasourceConfigPageParam.getProvider())) {
            queryWrapper.lambda().eq(AiDatasourceConfig::getProvider, aiDatasourceConfigPageParam.getProvider());
        }
        if(ObjectUtil.isAllNotEmpty(aiDatasourceConfigPageParam.getSortField(), aiDatasourceConfigPageParam.getSortOrder())) {
            CommonSortOrderEnum.validate(aiDatasourceConfigPageParam.getSortOrder());
            queryWrapper.orderBy(true, aiDatasourceConfigPageParam.getSortOrder().equals(CommonSortOrderEnum.ASC.getValue()),
                    StrUtil.toUnderlineCase(aiDatasourceConfigPageParam.getSortField()));
        } else {
            queryWrapper.lambda().orderByAsc(AiDatasourceConfig::getId);
        }
        return this.page(CommonPageRequest.defaultPage(), queryWrapper);
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void add(AiDatasourceConfigAddParam aiDatasourceConfigAddParam) {
        AiDatasourceConfig aiDatasourceConfig = BeanUtil.toBean(aiDatasourceConfigAddParam, AiDatasourceConfig.class);
        this.save(aiDatasourceConfig);
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void edit(AiDatasourceConfigEditParam aiDatasourceConfigEditParam) {
        AiDatasourceConfig aiDatasourceConfig = this.queryEntity(aiDatasourceConfigEditParam.getId());
        BeanUtil.copyProperties(aiDatasourceConfigEditParam, aiDatasourceConfig);
        this.updateById(aiDatasourceConfig);
    }

    @Transactional(rollbackFor = Exception.class)
    @Override
    public void delete(List<AiDatasourceConfigIdParam> aiDatasourceConfigIdParamList) {
        // 执行删除
        this.removeByIds(CollStreamUtil.toList(aiDatasourceConfigIdParamList, AiDatasourceConfigIdParam::getId));
    }

    @Override
    public AiDatasourceConfig detail(AiDatasourceConfigIdParam aiDatasourceConfigIdParam) {
        return this.queryEntity(aiDatasourceConfigIdParam.getId());
    }

    @Override
    public AiDatasourceConfig queryEntity(String id) {
        AiDatasourceConfig aiDatasourceConfig = this.getById(id);
        if(ObjectUtil.isEmpty(aiDatasourceConfig)) {
            throw new CommonException("数据源配置表不存在，id值为：{}", id);
        }
        return aiDatasourceConfig;
    }
}
